Y-net: a reducing gaussian noise convolutional neural network for MRI brain tumor classification with NADE concatenation

Biomed Phys Eng Express. 2021 Jul 13;7(5). doi: 10.1088/2057-1976/ac107b.

Abstract

Brain tumors are among the most serious cancers that can have a negative impact on a person's quality of life. The magnetic resonance imaging (MRI) analysis detects abnormal cell growth in the skull. Recently, machine learning models such as artificial neural networks have been used to detect brain tumors more quickly. To classify brain tumors, this research introduces the Y-net, a new convolutional neural network (CNN) based on the convolutional U-net architecture. We apply a NADE concatenation method in pre-processing the MR images for enhanced Y-net performance. We put our approach to the test using two MRI datasets of brain tumors. The first dataset contains three different types of brain tumors, while the second dataset includes a separate category for healthy brains. We show that our model is resistant to white noise and can obtain excellent classification accuracy with a limited number of medical images.

Keywords: NADE; U-net; Y-net; brain tumor classification; convolutional neural network.

MeSH terms

  • Brain Neoplasms* / diagnostic imaging
  • Humans
  • Magnetic Resonance Imaging
  • Neural Networks, Computer*
  • Neuroimaging
  • Quality of Life